'''
Author: huangwei barry.huangw@gmail.com
Date: 2024-12-23 09:53:26
LastEditors: huangwei barry.huangw@gmail.com
LastEditTime: 2024-12-23 19:57:42
FilePath: /pyqt6/test.py
Description: test code
'''
from yolov5.infer import Infer
import cv2
import torch

pt = "model\L.pt"
image = cv2.imread("pede.jpeg")
detect_person = Infer(pt)
detect_person.infer_image(image)
results = detect_person.parse_result()

print(detect_person.get_alert_info())
result_image = detect_person.plot_pred()
cv2.imwrite("result.png",result_image)
print(results)


# rtsp://admin:bjkjdx12@192.168.31.101:554
# rtsp://admin:bjkjdx12@192.168.31.102:554
